如果要绘制三维图形,首先需要使用下面的语句导入相应的对象:
from mpl_toolkits.mplot3d import Axes3D
然后使用下面的两种方式之一声明要创建三维子图:
ax = fig.gca(projection='3d')
ax = plt.subplot(111, projection='3d')
接下来就可以使用ax的plot()方法绘制三维曲线、plot_surface()方法绘制三维曲面、scatter()方法绘制三维散点图或bar3d()方法绘制三维柱状图了。
在绘制三维图形时,至少需要指定x、y、z三个坐标轴的数据,然后再根据不同的图形类型指定额外的参数设置图形的属性。绘制三维曲面的方法plot_surface()语法如下:
plot_surface(X, Y, Z, *args, **kwargs)
其中常用的参数有:1)rstride和cstride分别控制x和y两个方向的步长,这决定了曲面上每个面片的大小;2)color用来指定面片的颜色;3)cmap用来指定面片的颜色映射表。
绘制三维散点图的方法scatter()语法如下:
scatter(xs, ys, zs=0, zdir='z', s=20, c=None, depthshade=True, *args, **kwargs)
其中常用的参数有:1)xs、ys、zs分别用来指定散点符号的x、y、z坐标,如果同时为标量则指定一个三点符号的坐标,如果同时为等长数组则指定一系列散点符号的坐标;2)s用来指定散点符号的大小,可以是标量或与xs等长的数组;3)表8-